# Copyright (c) OpenMMLab. All rights reserved.
# from .ade import ADE20KDataset
# from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
# from .chase_db1 import ChaseDB1Dataset
# from .cityscapes import CityscapesDataset
# from .coco_stuff import COCOStuffDataset
# from .custom import CustomDataset
# from .dark_zurich import DarkZurichDataset
# from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset,
#                                RepeatDataset)
# from .drive import DRIVEDataset
# from .hrf import HRFDataset
# from .isprs import ISPRSDataset
# from .loveda import LoveDADataset
# from .night_driving import NightDrivingDataset
# from .pascal_context import PascalContextDataset, PascalContextDataset59
# from .potsdam import PotsdamDataset
# from .stare import STAREDataset
# from .voc import PascalVOCDataset
#
# __all__ = [
#     'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
#     'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
#     'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
#     'PascalContextDataset59', 'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset',
#     'STAREDataset', 'DarkZurichDataset', 'NightDrivingDataset',
#     'COCOStuffDataset', 'LoveDADataset', 'MultiImageMixDataset',
#     'ISPRSDataset', 'PotsdamDataset'
# ]

# new version just for RS datasets
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
from .custom import CustomDataset
from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset,
                               RepeatDataset)
from .isprs import ISPRSDataset
from .loveda import LoveDADataset
from .potsdam import PotsdamDataset
from .uda_dataset import UDADataset
__all__ = [
    'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
    'DATASETS', 'build_dataset', 'PIPELINES',
    'LoveDADataset', 'MultiImageMixDataset',
    'ISPRSDataset', 'PotsdamDataset', 'UDADataset'
]
